import pandas as pd
import argparse
parser = argparse.ArgumentParser(description="function parameters of argparse")
parser.add_argument('-fp', '--file_path', default='')
parser.add_argument('-sp', '--save_path', default='')
parser.add_argument('-c', '--cishu', default='')
parser.add_argument('-ce', '--csv_or_excel', default='excel')
args = parser.parse_args()

def xlsx_data_extension(file_path: str, beishu: int):
    dataframe = pd.read_excel(io=file_path)
    dataframe1 = dataframe
    for _ in range(beishu):
        dataframe = dataframe.append(dataframe1)
    return dataframe

def save_data(save_path: str, dataframe: pd.DataFrame, csv_or_excel: str):
    if csv_or_excel == 'excel':
        dataframe.to_excel(save_path)
    else:
        dataframe.to_csv(save_path, sep=',')

if __name__ == '__main__':
    save_data(args.save_path, xlsx_data_extension(args.file_path, int(args.cishu)), args.csv_or_excel)